import pytest
from weed_shop.core.models.dto import DiscountDTO, ProductDTO, ProductWithQuantityDTO


@pytest.fixture
def sample_discounts() -> list[DiscountDTO]:
    return [
        DiscountDTO(
            id=1, category_id=1, shop_id=1, required_quantity=3, discount_percent=10, works_inside_category=True,
        ),
    ]


@pytest.fixture
def sample_products() -> list[ProductWithQuantityDTO]:
    return [
        ProductWithQuantityDTO(ProductDTO(id=1, price=120, category_id=1), quantity=2),
        ProductWithQuantityDTO(ProductDTO(id=2, price=100, category_id=1), quantity=1),
    ]
